package com.ny.util;

import java.util.Random;

/**
 * 验证码生成工具类
 */
public class VerificationCodeUtil {

    /**
     * 生成指定长度的数字验证码
     * @param length 验证码长度（如6）
     * @return 数字验证码
     */
    public static String generateCode(int length) {
        if (length <= 0) {
            throw new IllegalArgumentException("验证码长度必须大于0");
        }
        Random random = new Random();
        StringBuilder code = new StringBuilder();
        for (int i = 0; i < length; i++) {
            code.append(random.nextInt(10)); // 生成0-9的随机数
        }
        return code.toString();
    }
}